import requests


def wechat_login(code):
    if code:
        app_id = 'your_app_id'  # 替换为你的微信小程序 appId
        app_secret = 'your_app_secret'  # 替换为你的微信小程序 appSecret

        url = 'https://api.weixin.qq.com/sns/jscode2session'
        params = {
            'appid': app_id,
            'secret': app_secret,
            'js_code': code,
            'grant_type': 'authorization_code'
        }

        response = requests.get(url, params=params)
        data = response.json()

        if 'openid' in data:# 判断是否获取到 openid 和 session_key
            openid = data['openid']
            session_key = data['session_key']
            # 在这里处理 openid 和 session_key，例如保存到数据库
            return {'openid': openid, 'session_key': session_key}
        else:
            return {'error': '获取 openid 和 session_key 失败'}


    return {'error': '请求方法不支持'}
